Utilizing a dataclass and enum together